from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `xm_model_pro` ADD `piclist` LONGTEXT NOT NULL;
        ALTER TABLE `xm_model_pro` ADD `price` DECIMAL(10,2) NOT NULL  DEFAULT 0;"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `xm_model_pro` DROP COLUMN `piclist`;
        ALTER TABLE `xm_model_pro` DROP COLUMN `price`;"""
